package leetcode101.mathematical_problem;

import java.util.HashSet;

/**
 * @author Synhard
 * @version 1.0
 * @Class Code15
 * @Description 202. 快乐数
 * 编写一个算法来判断一个数 n 是不是快乐数。
 *
 * 「快乐数」定义为：
 *
 * 对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和。
 * 然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。
 * 如果 可以变为  1，那么这个数就是快乐数。
 * 如果 n 是快乐数就返回 true ；不是，则返回 false 。
 *
 *  
 *
 * 示例 1：
 *
 * 输入：19
 * 输出：true
 * 解释：
 * 1^2 + 9^2 = 82
 * 8^2 + 2^2 = 68
 * 6^2 + 8^2 = 100
 * 1^2 + 0^2 + 0^2 = 1
 * 示例 2：
 *
 * 输入：n = 2
 * 输出：false
 *
 * @tel 13001321080
 * @email 823436512@qq.com
 * @date 2021-04-28 11:01
 */
public class Code15 {
    public static void main(String[] args) {
        System.out.println(isHappy(7777));
    }

    public static boolean isHappy(int n) {
        if (n == 1) {
            return true;
        }
        HashSet<Integer> hashSet = new HashSet<>();
        hashSet.add(n);
        while (true) {
            n = getSum(n);
            if (hashSet.contains(n)) {
                return false;
            }
            if (n == 1) {
                return true;
            }
            hashSet.add(n);
        }
    }

    private static int getSum(int n) {
        int res = 0;
        while (n != 0) {
            res += (n % 10) * (n % 10);
            n /= 10;
        }
        return res;
    }
}
